#!/bin/bash
# train tiantai 
python main.py --train_data ../data/tiantai_train --test_data ../data/tiantai_test \
     --epoches 30 --model_save_path models --batch_size 32  --lr 1e-4 \
     --train_list tiantai_train.txt --test_list tiantai_test.txt --gpu 3 --isTrain 1 \
     --model_save_name tiantai.pth
# test tiantai
python main.py  --test_data ../data/tiantai_test --batch_size 32  \
     --test_list tiantai_test.txt --gpu 3 --isTrain 0 \
     --pretrained_model models/tiantai.pth

# train with a exiting model
# be aware of param --pretrained_model and --model_save_name
#python main.py --train_data ../data/selected_rec --test_data ../data/selected_rec \
#        --epoches 1 --model_save_path models --batch_size 32  --lr 1e-4 \
#        --train_list train_1216.txt --test_list test_1216.txt --gpu 0 --isTrain 1 \
#        --pretrained_model models/model.pth --model_save_name fined.pth
